package com.cencat.user.mapper;

import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.cencat.user.entity.UserRole;
import com.cencat.user.entity.Role;
import com.cencat.user.dto.UserRoleQueryDTO;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;
import org.apache.ibatis.annotations.Select;

import java.util.List;

/**
 * 用户角色Mapper接口
 * @author cencat
 * @since 2024-01-01
 */
@Mapper
public interface UserRoleMapper extends BaseMapper<UserRole> {

    /**
     * 根据用户ID查询用户角色关联
     * @param userId 用户ID
     * @param tenantId 租户ID
     * @return 用户角色关联列表
     */
    @Select("SELECT * FROM user_role WHERE user_id = #{userId} AND tenant_id = #{tenantId}")
    List<UserRole> selectByUserId(@Param("userId") Long userId, @Param("tenantId") Long tenantId);

    /**
     * 根据角色ID查询用户角色关联
     * @param roleId 角色ID
     * @param tenantId 租户ID
     * @return 用户角色关联列表
     */
    @Select("SELECT * FROM user_role WHERE role_id = #{roleId} AND tenant_id = #{tenantId}")
    List<UserRole> selectByRoleId(@Param("roleId") Long roleId, @Param("tenantId") Long tenantId);

    /**
     * 根据用户ID删除用户角色关联
     * @param userId 用户ID
     * @param tenantId 租户ID
     * @return 删除数量
     */
    @Select("DELETE FROM user_role WHERE user_id = #{userId} AND tenant_id = #{tenantId}")
    int deleteByUserId(@Param("userId") Long userId, @Param("tenantId") Long tenantId);

    /**
     * 根据角色ID删除用户角色关联
     * @param roleId 角色ID
     * @param tenantId 租户ID
     * @return 删除数量
     */
    @Select("DELETE FROM user_role WHERE role_id = #{roleId} AND tenant_id = #{tenantId}")
    int deleteByRoleId(@Param("roleId") Long roleId, @Param("tenantId") Long tenantId);

    /**
     * 批量插入用户角色关联
     * @param userRoles 用户角色关联列表
     * @return 插入数量
     */
    int batchInsertUserRoles(@Param("userRoles") List<UserRole> userRoles);

    /**
     * 根据用户ID和角色ID列表删除用户角色关联
     * @param userId 用户ID
     * @param roleIds 角色ID列表
     * @param tenantId 租户ID
     * @return 删除数量
     */
    int deleteByUserIdAndRoleIds(@Param("userId") Long userId, @Param("roleIds") List<Long> roleIds, @Param("tenantId") Long tenantId);

    /**
     * 批量插入用户角色关联
     * 
     * @param userRoles 用户角色关联列表
     * @return 插入数量
     */
    int batchInsert(@Param("userRoles") List<UserRole> userRoles);

    /**
     * 根据用户ID列表查询角色ID列表
     * 
     * @param userIds 用户ID列表
     * @param tenantId 租户ID
     * @return 角色ID列表
     */
    List<Long> selectRoleIdsByUserIds(@Param("userIds") List<Long> userIds, @Param("tenantId") Long tenantId);

    /**
     * 统计用户角色关联数量
     * 
     * @param tenantId 租户ID
     * @return 关联数量
     */
    Long countUserRoles(@Param("tenantId") Long tenantId);

    /**
     * 统计用户的角色数量
     * 
     * @param userId 用户ID
     * @param tenantId 租户ID
     * @return 角色数量
     */
    Long countRolesByUserId(@Param("userId") Long userId, @Param("tenantId") Long tenantId);

    /**
     * 统计角色的用户数量
     * 
     * @param roleId 角色ID
     * @param tenantId 租户ID
     * @return 用户数量
     */
    Long countUsersByRoleId(@Param("roleId") Long roleId, @Param("tenantId") Long tenantId);

    /**
     * 根据用户ID和角色ID查询用户角色关联
     * 
     * @param userId 用户ID
     * @param roleId 角色ID
     * @param tenantId 租户ID
     * @return 用户角色关联
     */
    UserRole selectByUserIdAndRoleId(@Param("userId") Long userId, @Param("roleId") Long roleId, @Param("tenantId") Long tenantId);

    /**
     * 检查用户是否拥有指定角色
     * 
     * @param userId 用户ID
     * @param roleId 角色ID
     * @param tenantId 租户ID
     * @return 是否拥有角色
     */
    Boolean checkUserHasRole(@Param("userId") Long userId, @Param("roleId") Long roleId, @Param("tenantId") Long tenantId);

    /**
     * 检查用户是否拥有指定角色编码
     * 
     * @param userId 用户ID
     * @param roleCode 角色编码
     * @param tenantId 租户ID
     * @return 是否拥有角色
     */
    Boolean checkUserHasRoleCode(@Param("userId") Long userId, @Param("roleCode") String roleCode, @Param("tenantId") Long tenantId);

    /**
     * 根据角色ID列表查询用户ID列表
     * @param roleIds 角色ID列表
     * @param tenantId 租户ID
     * @return 用户ID列表
     */
    List<Long> selectUserIdsByRoleIds(@Param("roleIds") List<Long> roleIds, @Param("tenantId") Long tenantId);
    
    /**
     * 根据用户ID查询用户角色关联
     * @param userId 用户ID
     * @param tenantId 租户ID
     * @return 用户角色关联列表
     */
    List<UserRole> selectUserRoleByUserId(@Param("userId") Long userId, @Param("tenantId") Long tenantId);
    
    /**
     * 根据用户ID查询角色列表
     * @param userId 用户ID
     * @param tenantId 租户ID
     * @return 角色列表
     */
    List<Role> selectRolesByUserId(@Param("userId") Long userId, @Param("tenantId") Long tenantId);

    /**
     * 批量更新用户角色
     * @param userRoles 用户角色列表
     * @return 更新数量
     */
    int batchUpdateUserRoles(@Param("userRoles") List<UserRole> userRoles);

    /**
     * 分页查询用户角色关联
     * @param page 分页参数
     * @param queryDTO 查询条件
     * @return 用户角色关联列表
     */
    IPage<UserRole> selectUserRolePage(IPage<UserRole> page, @Param("query") UserRoleQueryDTO queryDTO);

    /**
     * 查询用户角色关联列表
     * @param queryDTO 查询条件
     * @return 用户角色关联列表
     */
    List<UserRole> selectUserRoleList(@Param("query") UserRoleQueryDTO queryDTO);

}